from mapper.testcase_mapper import TestcaseMapper
from model.Testcase_model import TestcaseModel


testcase_mapper = TestcaseMapper()


class TestcaseService:
    """
        测试用例服务层
    """

    def save(self, testcase_entity: TestcaseModel):
        """
        1、查询用例，是否存在，如果存在，则不新增
        2、反之就做新增操作
        :return:
        """
        if self.get(testcase_entity.id):
            return False
        else:
            testcase_mapper.save(testcase_entity)
            return True

    def delete(self, testcase_id):
        if self.get(testcase_id):
            testcase_mapper.delete(testcase_id)
            return True
        else:
            return False

    def get(self, testcase_id):
        return testcase_mapper.get(testcase_id)

    def get_all(self):
        return testcase_mapper.get_all()

    def update(self, update_testcase: dict):
        testcase_id = update_testcase.get("id")
        if self.get(testcase_id):
            testcase_mapper.update(testcase_id, update_testcase)
            return True
        else:
            return False
